tensor([[0.1000, 0.9800, 0.1000, 0.9800, 0.1000],
        [0.9000, 0.0200, 0.9000, 0.0200, 0.9000]], grad_fn=<RoundBackward1>)